package com.it.data_tree.tree.binaryTree;

import java.util.LinkedList;
import java.util.Queue;

/**
 * @Author: Coke
 * @DateTime: 2023/04/15/18:37
 * @注释: 二叉树最小深度
 **/
public class MinDepth {
    
    /**
     * @作者: Coke
     * @注释: 二叉树最小深度（解法1 - 递归）
     * @DateTime: 2023/4/16 13:46
     *
     * @param node:
     * @return int
     */
    public static int solution1(Node node) {
        if (node == null) {
            return 0;
        }
        int left = solution1(node.getLeft());
        int right = solution1(node.getRight());
        // 如果左子树为null,则以右子树为准
        if (left == 0){
            return right + 1;
        }
        // 如果右子树为null,则以左子树为准
        if (right == 0){
            return left + 1;
        }
        return  Integer.min(left, right) + 1;
    }
    
    /**
     * @作者: Coke
     * @注释:二叉树最小深度（解法2 - 层序遍历 - 最小层数为最小深度）
     * @DateTime: 2023/4/16 13:58
     *
     * @param node:
     * @return int
     */
    // 层序遍历遇到的第一个子节点所在的层就是最小深度
    public static int solution2(Node node){
        if (node == null){
            return 0;
        }
        // 使用存储临时节点元素
        Queue<Node> queue = new LinkedList<>();
        queue.offer(node);
        int min = 0;
        while (!queue.isEmpty()){
            int size = queue.size();
            min ++;
            for (int i = 0; i < size; i++) {
                Node poll = queue.poll();
                if (poll.getLeft() == null && poll.getRight() == null){
                    return min;
                }
                if (poll.getLeft() !=null){
                    queue.offer(poll.getLeft());
                }
                if(poll.getRight() != null){
                    queue.offer(poll.getRight());
                }
            }
        }
        return min;
    }
}
